from ultralytics import YOLO

# 加载训练好的模型权重
model = YOLO('./runs/detect/exp/weights/best.pt')

# 新图片文件夹的路径
folder_path = './datasets/images'

# 对文件夹内的所有图片进行推理
results = model.predict(
    source=folder_path,  # 新图片文件夹的路径
    conf=0.45,          # 置信度阈值
    iou=0.6,            # IOU阈值
    imgsz=640,          # 输入图像大小
    save_txt=True       # 是否保存标签信息
)
